Deep ensemble model for segmenting microscopy images in the presence of limited labeled data

Jan Mikolaj Kaminski, Ilary Allodi, Roser Montañana-Rosell, Raghavendra Selvan, Ole Kiehn

Keywords: ensemble, segmentation, microscopy

TL;DR: We propose deep ensemble model to perform microscopy image segmentation with limited labeled data.

Abstract: Obtaining large amounts of high quality labeled microscopy data is expensive and time-consuming. To overcome this issue, we propose a deep ensemble model which aims to utilise limited labeled training data. We train multiple identical Convolutional Neural Network (CNN) segmentation models on training data that is partitioned into folds in two steps. First, the data is split based on sample diversity or expert knowledge reflecting the possible {\em modes} of the underlying data distribution. In the second step, these partitions are split into random folds like in a cross-validation setting. Segmentation models based on the U-net architecture are trained on each of these resulting folds yielding the candidate models for our deep ensemble model which are aggregated to obtain the final prediction. The proposed deep ensemble model is compared to relevant baselines, in their ability to segment interneurons in microscopic images of mice spinal cord, showing improved performance on an independent test set.

Paper Type: methodological development

Primary Subject Area: Learning with Noisy Labels and Limited Data

Secondary Subject Area: Segmentation

Paper Status: original work, not submitted yet

Source Code Url: https://github.com/JanKaminski95/EnsembleNet.git

Data Set Url: Data set is internal and can be accessed upon request.

Registration: I acknowledge that publication of this at MIDL and in the proceedings requires at least one of the authors to register and present the work during the conference.

Authorship: I confirm that I am the author of this work and that it has not been submitted to another publication before.